package com.dss.code_generation.util;

import cn.hutool.cache.CacheUtil;
import cn.hutool.cache.impl.CacheObj;
import cn.hutool.cache.impl.TimedCache;
import cn.hutool.core.date.DateUnit;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;

/**
 * spring cache 工具类
 *
 * @author lenovo
 **/
@SuppressWarnings(value = {"unchecked", "rawtypes"})
@Component
public class CacheUtils {
    @Autowired

    /**
     * 默认缓存时长 60s
     */
    private static final Long DEFAULT_TIMEOUT = DateUnit.SECOND.getMillis() * 60;

    /**
     * 默认清理间隔时间 单位s
     */
    private static final Long CLEAN_TIMEOUT = DateUnit.SECOND.getMillis() * 1;

    /**
     * 全局时间超时缓存，构造器启动每秒检测。
     */
    public static TimedCache<String, Object> timedCache = CacheUtil.newTimedCache(DEFAULT_TIMEOUT);

    /**
     * 缓存基本的对象，Integer、String、实体类等
     *
     * @param key   缓存的键值
     * @param value 缓存的值
     */
    public <T> void setCacheObject(final String key, final T value) {
        timedCache.put(key, value);
    }

    /**
     * 缓存基本的对象，Integer、String、实体类等
     *
     * @param key      缓存的键值
     * @param value    缓存的值
     * @param timeout  时间
     */
    public <T> void setCacheObject(final String key, final T value, final Integer timeout) {
        timedCache.put(key,value,timeout);
    }


    /**
     * 判断 key是否存在
     *
     * @param key 键
     * @return true 存在 false不存在
     */
    public Boolean hasKey(String key) {
        return timedCache.containsKey(key);
    }

    /**
     * 获得缓存的基本对象。
     *
     * @param key 缓存键值
     * @return 缓存键值对应的数据
     */
    public <T> T getCacheObject(final String key) {
        Object o = timedCache.get(key);
        return (T)o;
    }

    /**
     * 删除单个对象
     *
     * @param key
     */
    public boolean deleteObject(final String key) {
        timedCache.remove(key);
        return true;
    }

    /**
     * 删除集合对象
     *
     * @param collection 多个对象
     * @return
     */
    public long deleteObject(final Collection collection) {
        for (Object o : collection) {
            timedCache.remove(o.toString());
        }
        return collection.size();
    }

    /**
     * 获得缓存的基本对象列表
     *
     * @param pattern 字符串前缀
     * @return 对象列表
     */
    public Collection<String> keys(final String pattern) {
        List<String> list = new ArrayList<>();
        Iterator<CacheObj<String, Object>> iterator = timedCache.cacheObjIterator();
        if (iterator.hasNext()){
            CacheObj<String, Object> obj = iterator.next();
            String key = obj.getKey();
            if (key.startsWith(pattern)){
                list.add(key);
            }
        }
        return list;
    }
}
